AI augmented risk-stratification